# -*- coding: utf-8 -*-
#### 人脸检测并识别计数 ####
# 依赖于ultralytics库
# 作者：Ekaros
# 版本：1.0




import re

def split_plus(text):
    # 定义一个空列表，用于存储分割后的字符串
    result = []
    # 定义一个空字符串，用于存储当前正在处理的字符串
    result = re.split('[^\u4e00-\u9fa5a-zA-Z0-9]', text)
    return result

# 分解晒单信息(细分业务类型)
# 输入：msg:str  晒单信息
def shaidan_xifen(msg):
    msg_sd = {}
    dict_list = ['战队','业务','号码','发展人']
    msg_list = msg.split('\n')
    for i in range(len(dict_list)):
        if i > 0:
            if ':' in msg_list[i]:
                if i == 1:
                    split_by = msg_list[i].split(':')[1]
                    msg_sd[dict_list[i]] = split_plus(split_by)
                else:
                    msg_sd[dict_list[i]] = msg_list[i].split(':')[1]
            elif '：' in msg_list[i]:
                msg_sd[dict_list[i]] = msg_list[i].split('：')[1]
        else:
            msg_sd[dict_list[i]] = msg_list[i]
    return msg_sd


# 分解晒单信息(不细分业务类型)
# 输入：msg:str  晒单信息
def shaidan(msg:str):
    msg_sd = {}
    dict_list = ['战队','业务','号码','发展人']
    msg_list = msg.split('\n')
    for i in range(len(dict_list)):
        if i > 0:
            if ':' in msg_list[i]:
                msg_sd[dict_list[i]] = msg_list[i].split(':')[1]
            elif '：' in msg_list[i]:
                msg_sd[dict_list[i]] = msg_list[i].split('：')[1]
        else:
            msg_sd[dict_list[i]] = msg_list[i]
    return msg_sd
